import pandas as pd
import matplotlib.pyplot as plt

# 读取数据
df = pd.read_excel('柱状图-某公司产品销售数据.xlsx')

# 按地区筛选数据
df1 = df[df['地区'] == '北京市']
df2 = df[df['地区'] == '上海市']
df3 = df[df['地区'] == '武汉市']

print(df1)

# 提取季度数据
x1 = df1['季度']
x2 = df2['季度']
x3 = df3['季度']

# 设置图表参数
plt.figure(figsize=(26, 18))  # 调整画布大小
plt.rcParams['font.sans-serif'] = 'SimHei'  # 设置中文显示
plt.title('某公司1-2月实际支出堆积柱状图')  # 设置标题
plt.ylabel('销售额（万元）')  # 设置y轴标签
width = 0.4  # 设置柱子宽度

# 绘制堆积柱状图
plt.bar(x1, df1['销售额（万元）'], width, color='green')
plt.bar(x2, df2['销售额（万元）'], width, bottom=df1['销售额（万元）'].values, color='blue')
plt.bar(x3, df3['销售额（万元）'], width, bottom=df1['销售额（万元）'].values + df2['销售额（万元）'].values, color='red')

# 设置图例
plt.legend(('北京', '上海', '武汉'))

# 显示图表
plt.show()
